import pymysql

from .config import ConfigParser


class MySqlClient:

    def __init__(self):
        config = ConfigParser()
        self.db_profile = config.get_mysql_profile()

    def connect(self):
        """
        连接数据库
        :return:
        """
        return pymysql.connect(**self.db_profile)

    def execute_sql(self, sql):
        """
        执行sql语句
        :param sql:
        :return: 返回一个元组
        """
        # 1. 创建数据库连接
        con = self.connect()
        # 2. 创建游标
        cs = con.cursor()
        # 3. 执行sql语句
        cs.execute(sql)
        # 4. 取值
        # all：获取所有值
        # cs.fetchall()
        # one：只取一条数据
        value = cs.fetchone()
        # 关闭游标
        cs.close()
        # 关闭连接
        con.close()
        return value
